from selenium.webdriver.common.by import By

from pom.base import BaseDriver


class MainPage(BaseDriver):

    _login_link = By.LINK_TEXT,"登录"
    _register_link = By.LINK_TEXT,"注册"

    def login_link(self):
        return self.driver.find_element(*self._login_link)

    def register_link(self):
        return self.driver.find_element(*self._register_link)